package cn.itcast.bos.service.base.impl;


import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.criteria.CriteriaQuery;
import javax.persistence.criteria.Predicate;
import javax.persistence.criteria.Root;

import org.apache.commons.lang.StringUtils;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.Row;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.domain.Specification;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import cn.itcast.bos.dao.AreaDao;
import cn.itcast.bos.domain.base.Area;
import cn.itcast.bos.service.base.AreaService;
import cn.itcast.bos.util.PinYin4jUtils;
@Transactional
@Service
public class AreaServiceImpl implements AreaService {
    @Autowired
    private AreaDao areaDao;
    @Override
    public void saveBatch(File file, String fileFileName) throws IOException {
        //编写解析代码逻辑
		//给予/xls格式解析HSSF
		//1.加载Excel文件对象
		HSSFWorkbook hssfWorkbook = new HSSFWorkbook(new FileInputStream(file));
		//2.读取一个sheet
		HSSFSheet sheet = hssfWorkbook.getSheetAt(0);
		List<Area> list = new ArrayList();
		//3.读取sheet中的每一行
		for (Row row :sheet) {
			//一行对应一个区域对象
            if (row.getRowNum()==0){
                //第一行跳过
                continue;
            }
            //跳过空行
            if(row.getCell(0)==null ||
                    StringUtils.isBlank(row.getCell(0).getStringCellValue())){
                continue;
            }
            Area area = new Area();
            area.setId(row.getCell(0).getStringCellValue());
            area.setProvince(row.getCell(1).getStringCellValue());
            area.setCity(row.getCell(2).getStringCellValue());
           area.setDistrict(row.getCell(3).getStringCellValue());
            area.setPostcode(row.getCell(4).getStringCellValue());

            //基于pinyin4j生成城市编码和简码
            String province = area.getProvince();
            String city = area.getCity();
            String district = area.getDistrict();
            province = province.substring(0,province.length()-1);
            city = city.substring(0,city.length()-1);
            district = district.substring(0,district.length()-1);
            //简码
            String[] headArray = PinYin4jUtils.getHeadByString(province+city+district);
            StringBuffer buffer = new StringBuffer();
            for (String headStr : headArray){
                buffer.append(headStr);
            }
            String shortcode = buffer.toString();
            area.setShortcode(shortcode);
            //城市编码
            String citycode = PinYin4jUtils.hanziToPinyin(city,"");
            area.setCitycode(citycode);

		    list.add(area);
		}

		areaDao.save(list);

    }

    @Override
    public Page<Area> pageQuerry(Pageable pageable, Area area) {
        Specification<Area> specification = new Specification<Area>() {
            @Override
            public Predicate toPredicate(Root<Area> root, CriteriaQuery<?> query, CriteriaBuilder cb) {
                List<Predicate> list = new ArrayList<>();
                if (StringUtils.isNotBlank(area.getProvince())) {
                    Predicate p1 = cb.like(root.get("province").as(String.class), "%" + area.getProvince() + "%");
                    list.add(p1);
                }
                if (StringUtils.isNotBlank(area.getCity())) {
                    Predicate p2 = cb.like(root.get("city").as(String.class), "%" + area.getCity() + "%");
                    list.add(p2);
                }
                if (StringUtils.isNotBlank(area.getDistrict())) {
                    Predicate p3 = cb.like(root.get("district").as(String.class), "%" + area.getDistrict() + "%");
                    list.add(p3);
                }

                return cb.and(list.toArray(new Predicate[list.size()]));
            }
        };
        return areaDao.findAll(specification,pageable);
    }

}
